import xlrd


class BankCardData:

    # 正常数据
    def getSuccessData(self):
        workBook = xlrd.open_workbook(filename='../TestData/test_bankCard_data.xls')
        sheet = workBook.sheet_by_name('Sheet1')
        arr = []
        for i in range(1, sheet.nrows):
            e = sheet.row_values(i)
            if type(e[1]) is float:
                e[1] = str(int(e[1]))
            e[4] = e[4].split(',')
            arr.append(e)

        return arr

    # 异常卡号
    def getBankNumberErrorData(self):
        workBook = xlrd.open_workbook(filename='../TestData/test_bankCard_data.xls')
        sheet = workBook.sheet_by_name('Sheet2')
        arr = []
        for i in range(1, sheet.nrows):
            e = sheet.row_values(i)
            if type(e[1]) is float:
                e[1] = str(int(e[1]))
            if type(e[4]) is float:
                e[4] = int(e[4])
            arr.append(e)
        return arr
    # 卡号重复
    def getReBankNumberErrorData(self):
        # 用户名
        # 用户登录密码
        # 银行
        # 卡号
        # 余额
        # 类型 借记卡 1 信用卡 2
        # 默认 还是 非 默认 默认为 1 非默认为0
        # 判断条件

        return self.read_data(sheet_name='Sheet3')

    # 借记卡 和 信用卡 冲突
    def getReBankNumberErrorData2(self):
        return self.read_data(sheet_name='Sheet4')

    #  删除卡 数据
    def getDelBankCardSuccessData(self):
        return self.read_data(sheet_name='Sheet5')

    # 删除卡失败
    def getDelBankCardFailData(self):
        return self.read_data(sheet_name='Sheet6')

    # 修改银行卡 名
    def getModifyBankNameData(self):
        return self.read_data(sheet_name='Sheet7')

    # 修改银行卡 卡号
    def getModifyBankNumberData(self):
        return self.read_data(sheet_name='Sheet8')

    # 修改为异常卡号
    def getModifyBankNumberData2(self):
        return self.read_data(sheet_name='Sheet9')


    def getModifyBankCardTypeData(self):
        return self.read_data(sheet_name='Sheet10')


    def read_data(self,sheet_name):
        workBook = xlrd.open_workbook(filename='../TestData/test_bankCard_data.xls')
        sheet = workBook.sheet_by_name(sheet_name)
        arr = []
        for i in range(1, sheet.nrows):
            e = sheet.row_values(i)
            if type(e[1]) is float:
                e[1] = str(int(e[1]))
            if type(e[4]) is float:
                e[4] = str(int(e[4]))
            if type(e[5]) is float:
                e[5] = str(int(e[5]))
            if type(e[6]) is float:
                e[6] = str(int(e[6]))
            if type(e[7]) is float:
                e[7] = int(e[7])

            arr.append(e)

        return arr


if __name__ == '__main__':
    print(BankCardData().getModifyBankCardTypeData())
